Skip to content

Conversation

@amitchawla1
Copy link

@amitchawla1 amitchawla1 commented Oct 16, 2025

CollectiveBuilder takes ElementAccumulator, ElementCompute and ElementC as template input.
In test case, these datatypes are float, float and half_t.
Current code assumes all to be same (initiated with accumulator dtype), leading to datatype related assert in Copy.

This PR passes correct dataype for ElementC, ElementCompute is taken from FusionCallbacks operation and ElementAccumulator from ElementCompute.

@amitchawla1 amitchawla1 force-pushed the dev/achawla/cutlass_252 branch from c39f4ba to a391e80 Compare October 16, 2025 12:45
@petercad
Copy link

@amitchawla1 I think this issue is already addressed by #482.

Copy link

@jiyang1011 jiyang1011 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@rolandschulz
Copy link

Please explain in the PR description why the PR is fixing it the way it is (and not some other way e.g. #482).

@Antonyvance Antonyvance changed the title [CUTLASS-252] Epilogue DataType Mismatch Epilogue DataType Mismatch Oct 17, 2025
@Antonyvance Antonyvance added bug Something isn't working urgent PR requires a urgent attention (for release or blocking another PR) labels Oct 17, 2025
@Antonyvance Antonyvance added this to the 0.6 milestone Oct 17, 2025
Signed-off-by: Chawla, Amit K <[email protected]>
@amitchawla1 amitchawla1 force-pushed the dev/achawla/cutlass_252 branch from a391e80 to bf1764c Compare October 24, 2025 02:22
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

bug Something isn't working urgent PR requires a urgent attention (for release or blocking another PR)

Projects

None yet

Development

Successfully merging this pull request may close these issues.

5 participants